package fileUpload.optimized;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;

/*
    文件上传案例服务器端:读取客户端上传的文件,保存到服务器的硬盘,给客户端回写"上传成功"

    明确:
        数据源
        目的地:服务器的硬盘 d:\\upload\\1.jpg

    实现步骤:
        1.创建一个服务器ServerSocket对象,和系统要指定的端口号
        2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
        3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
        4.判断d:\\upload文件夹是否存在,不存在则创建
        5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
        6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
        7.使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
        8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
        9.使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功"
        10.释放资源(FileOutputStream,Socket,ServerSocket)
 */
public class TCPServer {
    public static void main(String[] args) {
        try (//1.创建一个服务器ServerSocket对象,和系统要指定的端口号
             ServerSocket server = new ServerSocket(8888);
        ) {
            /*
                让服务器一直处于监听状态(死循环accept方法)
                有一个客户端上传文件,就保存一个文件
            */
            while (true) {
                //2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
                Socket socket = server.accept();
                /*
                    使用多线程技术,提高程序的效率，使用了匿名内部类
                    有一个客户端上传文件,就开启一个线程,完成文件的上传
                */
                new Thread(new Runnable() {
                    public void run() {
                        try {
                            //3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
                            InputStream is = socket.getInputStream();
                            //4.判断文件夹是否存在,不存在则创建
                            File file = new File("Net/uploads");
                            if (!file.exists()) {
                                file.mkdirs();
                            }
                            //5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
                            /*
                                自定义一个文件的命名规则:防止同名的文件被覆盖
                                规则:域名+毫秒值
                            */
                            String filename = "anonyeast" + System.currentTimeMillis() + ".txt";
                            FileOutputStream fos = new FileOutputStream(file + "/" + filename);
                            //6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
                            byte[] bytes = new byte[1024];
                            int b;
                            while ((b = is.read(bytes)) != -1) {
                                //7.使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
                                fos.write(bytes, 0, b);
                            }
                            //8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
                            OutputStream os = socket.getOutputStream();
                            //9.使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功"
                            os.write("上传成功".getBytes());
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                }).start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}
